package com.yc.vote.util;



import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;

import java.util.Collections;

public class CodeGenerator {

    public static void main(String[] args) {
        FastAutoGenerator.create("jdbc:mysql:///db_vote?useSSL=false", "root", "yyyl")
                .globalConfig(builder -> {
                    builder.author("baomidou") // 设置作者
//                            用于生成API文档相关功能
//                            .enableSwagger() // 开启 swagger 模式
                            .fileOverride() // 覆盖已生成文件
                            .outputDir("D:\\java\\git\\git local\\s3\\vote\\code"); // 指定输出目录
                })
//                自定义的转换处理器
//.dataSourceConfig(builder -> builder.typeConvertHandler((globalConfig, typeRegistry, metaInfo) -> {
//            int typeCode = metaInfo.getJdbcType().TYPE_CODE;
//            if (typeCode == Types.SMALLINT) {
//                // 自定义类型转换
//                return DbColumnType.INTEGER;
//            }
//            return typeRegistry.getColumnType(metaInfo);
//
//        }))
//                设定生成的包路径
        .packageConfig(builder -> {
            builder.parent("com.yc") // 设置父包名
                    .moduleName("vote") // 设置父包模块名
                    .pathInfo(Collections.singletonMap(OutputFile.mapper, "D:\\java\\git\\git local\\s3\\vote\\code")); // 设置mapperXml生成路径
        }).strategyConfig(builder -> {
            builder//.addInclude("t_simple") // 设置需要生成的表名
                    .addTablePrefix("tp_"); // 设置过滤表前缀
        }).templateEngine(new FreemarkerTemplateEngine()).execute();// 使用Freemarker引擎模板，默认的是Velocity引擎模板
    }


}
